import os
import dotenv

dotenv.load_dotenv()

NEO4J_URI = os.getenv("NEO4J_URI")
NEO4J_USERNAME = os.getenv("NEO4J_USERNAME")
NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")

EMBEDDING_API_BASE = os.getenv("EMBEDDING_API_BASE")
EMBEDDING_API_KEY = os.getenv("EMBEDDING_API_KEY")
EMBEDDING_MODEL_NAME = os.getenv("EMBEDDING_MODEL_NAME")
EMBEDDING_DIM_SIZE = int(os.getenv("EMBEDDING_DIM_SIZE"))

AGGREGATION_MODEL_API_BASE = os.getenv("AGGREGATION_MODEL_API_BASE")
AGGREGATION_MODEL_API_KEY = os.getenv("AGGREGATION_MODEL_API_KEY")
AGGREGATION_MODEL_NAME = os.getenv("AGGREGATION_MODEL_NAME")

SESSION_MANAGER_CAPACITY: int = 1000